package com.woniuxy.util;

import com.woniuxy.bean.Menu;

import java.util.ArrayList;
import java.util.List;

public class MenuBuilder {
    /**
     * 用于构建具有层次关系的菜单
     * @param menus:没有层次关系的菜单集合
     * @param topId：顶级菜单的id
     * @return
     */
    public static List<Menu> build(List<Menu> menus, int topId){
        List<Menu> all=new ArrayList<Menu>();
        for(Menu n1:menus){
            //循环整个集合（没有层关系的集合），然后将一级菜单放入新的集合中
            if(n1.getPid()==topId){
                all.add(n1);
            }
            for(Menu n2:menus){
                if(n1.getId()==n2.getPid()){
                    n1.getChildren().add(n2);
                }
            }
        }
        return all;
    }
}
